package com.zh.oauth.util;

import com.zh.common.entity.ElecMenu;
import com.zh.oauth.entity.ElecAuth;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

/**
 * code is far away from bug with the animal protecting
 * ┏┓　　　┏┓
 * ┏┛┻━━━┛┻┓
 * ┃　　　　　　　┃
 * ┃　　　━　　　┃
 * ┃　┳┛　┗┳　┃
 * ┃　　　　　　　┃
 * ┃　　　┻　　　┃
 * ┃　　　　　　　┃
 * ┗━┓　　　┏━┛
 * 　　┃　　　┃神兽保佑
 * 　　┃　　　┃代码无BUG！
 * 　　┃　　　┗━━━┓
 * 　　┃　　　　　　　┣┓
 * 　　┃　　　　　　　┏┛
 * 　　┗┓┓┏━┳┓┏┛
 * 　　　┃┫┫　┃┫┫
 * 　　　┗┻┛　┗┻┛
 *
 * @Description: 构造菜单树
 * @Author: Li.HuiNeng
 * @Create in: 2019-09-26 9:57
 * @Version: 1.0.1
 */
public class MenuUtil {


    public static List<ElecMenu> getTreeData(List<ElecMenu> list) {
        List<ElecMenu> returnList = new ArrayList<ElecMenu>();
        for (Iterator<ElecMenu> iterator = list.iterator(); iterator.hasNext();) {
            ElecMenu res = iterator.next();
            //判断是否为第一个节点: 为0表示父节点
            if (res.getPid() == 0) {
                recursionFn(list, res);
                returnList.add(res);
            }
        }
        return returnList;
    }

    /**
     * 递归寻找子节点
     * @Title: recursionFn
     * @Description: 递归寻找子节点
     * @param beans 所有节点
     * @param current 当前节点
     */
    private static void recursionFn(List<ElecMenu> beans, ElecMenu current) {
        List<ElecMenu> childsList = getChildList(beans, current);
        current.setChildren(childsList);
        for (ElecMenu nextChild : childsList) {
            recursionFn(beans, nextChild);
        }
    }

    /**
     * 获得指定节点下的所有子节点
     * @Title: getChildList
     * @Description: 获得指定节点下的所有子节点
     * @param beans 所有节点集合
     * @param current 当前节点
     * @return
     */
    private static List<ElecMenu> getChildList(List<ElecMenu> beans, ElecMenu current) {
        List<ElecMenu> childsList = new ArrayList<ElecMenu>();
        Iterator<ElecMenu> it = beans.iterator();
        while (it.hasNext()) {
            ElecMenu child = it.next();
            if (current.getId().equals(child.getPid())) {
                childsList.add(child);
            }
        }
        return childsList;
    }
}
